from django.forms import ModelForm
from Proyectos.models import Proyecto, Link, Documento
from Grupos.models import Grupo
from django.contrib.auth.models import User

class ProyectoForm(ModelForm):
	def __init__(self, user, *args, **kwargs):
		super(ProyectoForm, self).__init__(*args, **kwargs)
		self.fields['creador'].queryset = User.objects.filter(pk = user.id)
		self.fields['creador'].empty_label=None
		self.fields['grupos'].queryset=Grupo.objects.filter(miembros__id__exact=user.id).order_by('nombre')

	class Meta:
		model = Proyecto
		exclude = ('links','documentos')

class ProyectoLinksOnlyForm(ModelForm):
	def __init__(self, idproyecto, *args, **kwargs):
		super(ProyectoLinksOnlyForm, self).__init__(*args, **kwargs)
		self.fields['links'].queryset=Link.objects.filter(proylinks__id__exact=idproyecto).order_by('titulo')

	class Meta:
		model = Proyecto
		exclude = ('nombre','imagen','creador','grupos','documentos')

class ProyectoDocsOnlyForm(ModelForm):
	def __init__(self, idproyecto, *args, **kwargs):
		super(ProyectoDocsOnlyForm, self).__init__(*args, **kwargs)
		self.fields['documentos'].queryset=Documento.objects.filter(proydocs__id__exact=idproyecto).order_by('titulo')

	class Meta:
		model = Proyecto
		exclude = ('nombre','imagen','creador','grupos','links')

class LinkForm(ModelForm):
	def __init__(self, *args, **kwargs):
		super(LinkForm, self).__init__(*args, **kwargs)

	class Meta:
		model = Link

class DocumentoForm(ModelForm):
	def __init__(self, *args, **kwargs):
		super(DocumentoForm, self).__init__(*args, **kwargs)

	class Meta:
		model = Documento
